Python Short Hand If Else